#!/bin/bash
# 使用Docker编译FFmpeg的脚本
# 使用方法: ./docker_build_ffmpeg.sh

set -e

# 颜色输出
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

print_info() {
    echo -e "${BLUE}ℹ️  $1${NC}"
}

print_success() {
    echo -e "${GREEN}✅ $1${NC}"
}

print_warning() {
    echo -e "${YELLOW}⚠️  $1${NC}"
}

print_error() {
    echo -e "${RED}❌ $1${NC}"
}

# 检查Docker是否安装
check_docker() {
    if ! command -v docker &> /dev/null; then
        print_error "Docker未安装，请先安装Docker Desktop"
        print_info "下载地址: https://www.docker.com/products/docker-desktop"
        exit 1
    fi
    
    if ! docker info &> /dev/null; then
        print_error "Docker未运行，请启动Docker Desktop"
        exit 1
    fi
    
    print_success "Docker环境检查通过"
}

# 构建Docker镜像
build_docker_image() {
    print_info "构建FFmpeg编译镜像..."
    
    docker build -f Dockerfile.ffmpeg -t ffmpeg-android-builder .
    
    if [ $? -eq 0 ]; then
        print_success "Docker镜像构建完成"
    else
        print_error "Docker镜像构建失败"
        exit 1
    fi
}

# 运行Docker编译
run_docker_build() {
    print_info "开始Docker编译FFmpeg..."
    
    # 创建输出目录
    mkdir -p ../android/app/src/main/jniLibs
    
    # 运行Docker容器
    docker run --rm \
        -v "$(pwd)/../android/app/src/main/jniLibs:/output" \
        ffmpeg-android-builder
    
    if [ $? -eq 0 ]; then
        print_success "Docker编译完成"
    else
        print_error "Docker编译失败"
        exit 1
    fi
}

# 验证编译结果
verify_build() {
    print_info "验证编译结果..."
    
    local output_dir="../android/app/src/main/jniLibs"
    local architectures=("arm64-v8a" "armeabi-v7a" "x86_64")
    
    for arch in "${architectures[@]}"; do
        local arch_dir="$output_dir/$arch"
        if [ -d "$arch_dir" ] && [ "$(ls -A $arch_dir)" ]; then
            print_success "$arch 架构编译成功"
        else
            print_warning "$arch 架构编译失败或文件为空"
        fi
    done
}

# 主函数
main() {
    print_info "开始Docker编译FFmpeg..."
    
    # 检查Docker环境
    check_docker
    
    # 构建Docker镜像
    build_docker_image
    
    # 运行编译
    run_docker_build
    
    # 验证结果
    verify_build
    
    print_success "FFmpeg Docker编译完成！"
    print_info "库文件位置: ../android/app/src/main/jniLibs"
    print_info ""
    print_info "Docker编译的优势:"
    print_info "  ✅ 环境一致性 - 任何平台都能获得相同结果"
    print_info "  ✅ 工具链完整 - 包含所有必需的编译工具"
    print_info "  ✅ 依赖管理 - 自动处理复杂的依赖关系"
    print_info "  ✅ 隔离性 - 不影响主机系统"
    print_info "  ✅ 可重复性 - 每次构建都能得到相同的结果"
}

# 执行主函数
main "$@"
